list comprehension solution

welcome to shbcf.ru